﻿/*******************************************************************************
 * Copyright (c) 2007
 * Sergey Berezin, Dmitry Voitsekhovskiy
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms,
 * with or without modification, are permitted provided
 * that the following conditions are met:
 *
 * - Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * - Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * - Neither the name of the Moscow State University nor the names
 * of its contributors may be used to endorse or promote products
 * derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
 * THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ******************************************************************************/
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace Faml.Operations
{
    [OperationContainer]
    public static class GenericSubarray
    {

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[] array, int[] indices)
        {
            T[] result = new T[indices.Length];
            for (int i = 0; i < indices.Length; i++)
            {
                result[i] = array[indices[i]];
            }

            return result;
        }


        [OperationName("subarray")]
        public static T Subarray<T>(T[] array, int index)
        {
            return array[index];
        }


        [OperationName("subarray")]
        public static T[,] Subarray<T>(T[,] array, int[] indicesX, int[] indicesY)
        {
            T[,] result = new T[indicesX.Length, indicesY.Length];
            for (int i = 0; i < indicesX.Length; i++)
                for (int j = 0; j < indicesY.Length; j++)
                {
                    result[i, j] = array[indicesX[i], indicesY[j]];
                }

            return result;
        }


        [OperationName("subarray")]
        public static T[] Subarray<T>(T[,] array, int indexX, int[] indicesY)
        {
            T[] result = new T[indicesY.Length];
            for (int j = 0; j < indicesY.Length; j++)
            {
                result[j] = array[indexX, indicesY[j]];
            }

            return result;
        }


        [OperationName("subarray")]
        public static T[] Subarray<T>(T[,] array, int[] indicesX, int indexY)
        {
            T[] result = new T[indicesX.Length];
            for (int j = 0; j < indicesX.Length; j++)
            {
                result[j] = array[indicesX[j], indexY];
            }

            return result;
        }


        [OperationName("subarray")]
        public static T Subarray<T>(T[,] array, int indexX, int indexY)
        {
            return array[indexX, indexY];
        }


        #region 3d subarrays

        [OperationName("subarray")]
        public static T[, ,] Subarray<T>(T[, ,] array, int[] indicesX, int[] indicesY, int[] indicesZ)
        {
            T[, ,] result = new T[indicesX.Length, indicesY.Length, indicesZ.Length];

            for (int i = 0; i < indicesX.Length; i++)
                for (int j = 0; j < indicesY.Length; j++)
                    for (int k = 0; k < indicesZ.Length; k++)
                    {
                        result[i, j, k] = array[indicesX[i], indicesY[j], indicesZ[k]];
                    }

            return result;
        }

        [OperationName("subarray")]
        public static T[,] Subarray<T>(T[, ,] array, int[] indicesX, int[] indicesY, int z)
        {
            T[,] result = new T[indicesX.Length, indicesY.Length];

            for (int i = 0; i < indicesX.Length; i++)
                for (int j = 0; j < indicesY.Length; j++)
                    result[i, j] = array[indicesX[i], indicesY[j], z];

            return result;
        }

        [OperationName("subarray")]
        public static T[,] Subarray<T>(T[, ,] array, int[] indicesX, int y, int[] indicesZ)
        {
            T[,] result = new T[indicesX.Length, indicesZ.Length];

            for (int i = 0; i < indicesX.Length; i++)
                for (int j = 0; j < indicesZ.Length; j++)
                    result[i, j] = array[indicesX[i], y, indicesZ[j]];

            return result;
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, ,] array, int[] indicesX, int y, int z)
        {
            T[] result = new T[indicesX.Length];

            for (int i = 0; i < indicesX.Length; i++)
                    result[i] = array[indicesX[i], y, z];

            return result;
        }


        [OperationName("subarray")]
        public static T[,] Subarray<T>(T[, ,] array, int x, int[] indicesY, int[] indicesZ)
        {
            T[,] result = new T[indicesY.Length, indicesZ.Length];

            for (int j = 0; j < indicesY.Length; j++)
                for (int k = 0; k < indicesZ.Length; k++)
                {
                    result[j, k] = array[x, indicesY[j], indicesZ[k]];
                }

            return result;
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, ,] array, int x, int[] indicesY, int z)
        {
            T[] result = new T[indicesY.Length];

            for (int j = 0; j < indicesY.Length; j++)
                result[j] = array[x, indicesY[j], z];

            return result;
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, ,] array, int x, int y, int[] indicesZ)
        {
            T[] result = new T[indicesZ.Length];

            for (int j = 0; j < indicesZ.Length; j++)
                result[j] = array[x, y, indicesZ[j]];

            return result;
        }

        [OperationName("subarray")]
        public static T Subarray<T>(T[, ,] array, int x, int y, int z)
        {
            return array[x, y, z];
        }

        #endregion

        #region 4d subarrays

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, , ,] array, int x, int y, int z, int[] w)
        {
            throw new NotImplementedException();
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, , ,] array, int[] x, int y, int z, int w)
        {
            throw new NotImplementedException();
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, , ,] array, int x, int[] y, int z, int w)
        {
            throw new NotImplementedException();
        }

        [OperationName("subarray")]
        public static T[] Subarray<T>(T[, , ,] array, int x, int y, int[] z, int w)
        {
            throw new NotImplementedException();
        }

        #endregion
    }
}
